#!/bin/bash
set -ex
#set -e
# -------------------------------------------------------------------------------
# Filename:     deploy_env_s.sh
# UpdateDate:   2025/07/10
# Description:  一键搭建验证环境。
# Example:      cd /home/share/pytorch2.5/comfyui/tools/ && bash deploy_env_s.sh
# Notes:        本脚本适用于初次进入docker容器环境；不适用于再次进入容器，可能会有意想不到问题
# -------------------------------------------------------------------------------

# 1. 切换到推理环境
source /torch/venv3/pytorch_infer/bin/activate
# 2.1. 安装 ComfyUI
cd /workspace/comfyui && bash test.sh
# 2.2. 安装 ComfyUI-Manager。ComfyUI Manager 是用于扩展和管理自定义节点、模型及插件的工具。
cd /workspace/comfyui/ComfyUI/custom_nodes
git clone https://github.com/ltdrdata/ComfyUI-Manager comfyui-manager
# 2.3. 安装 Diffusers
cd /workspace/diffusers/ && bash test.sh
# 2.4. 一键下载并生成本地缓存文件。在联网环境下运行一次以生成 .cache/*.json 本地文件。
cd /home/share/pytorch2.5/comfyui/tools && python fetch_comfyui_cache.py
# 2.5. 设置离线: 禁用联网、强制使用本地缓存
#python patch_local_cache.py
# 3. 启动 ComfyUI
#python main.py
# 如有需要，您可以使用 --listen 和 --port 选项更改这些设置。在启动 ComfyUI 时设置环境变量 export COMFYUI_MANAGER_NO_INTERNET=1,关闭联网更新行为
#export COMFYUI_MANAGER_NO_INTERNET=1
cd /workspace/comfyui/ComfyUI
python main.py --listen 0.0.0.0 --port 8188
